"""
python fujian2_groupByCategory.py
"""

import os
import pandas as pd
import json

def read_and_group_data(input_csv_path, output_dir):
    # 读取CSV文件，使用UTF-8编码
    df = pd.read_csv(input_csv_path, encoding='utf-8')

    # 按照类别进行分组
    grouped = df.groupby('Category')

    # 遍历每个组
    for category, group in grouped:
        # 按照日期升序排列
        group = group.sort_values(by='Date')

        # 创建要保存的JSON对象，字段名使用英文
        json_data = group[['Date', 'Sales']].rename(columns={'Date': 'date', 'Sales': 'sales'}).to_dict(orient='records')

        # 创建输出路径
        category_folder = os.path.join(output_dir, 'groupByCategory')
        os.makedirs(category_folder, exist_ok=True)  # 创建目录（如果不存在）

        # JSON文件名
        json_file_path = os.path.join(category_folder, f'category_{category}.json')

        # 将数据写入JSON文件
        with open(json_file_path, 'w', encoding='utf-8') as json_file:
            json.dump(json_data, json_file, ensure_ascii=False, indent=4)

        print(f"已保存: {json_file_path}")

# 示例用法
if __name__ == "__main__":
    input_csv = '../fujian/fujian2.csv'  # 使用相对路径
    output_directory = '../fujian/fujian2'  # 使用相对路径
    
    read_and_group_data(input_csv, output_directory)
